from rest_framework import serializers

from userinfo.models import Company


class AddPositionSerializer(serializers.Serializer):
    name = serializers.CharField()
    salary = serializers.CharField()


class CompanySerializer(serializers.ModelSerializer):
    class Meta:
        model = Company
        fields = ['name', 'logo', 'company_profile', 'create_time', 'start_time', 'end_time']
